# -*- coding: utf-8 -*-
"""
Created on Wed Jan  3 11:04:40 2024

@author: 85401
"""

# 导入PyMuPDF库
import fitz


# 定义一个名为split_pdf的函数，设置两个参数：
# source_filepath（我们要处理的PDF文件的路径）和 output_folder（分割以后输出文件夹的路径）
def split_pdf(source_filepath, output_folder):
    # 使用fitz.open方法打开源PDF文件，并将其对象赋值给source_pdf变量
    source_pdf = fitz.open(source_filepath)
    output_pdf = fitz.open()

    output_pdf.insert_pdf(source_pdf, from_page=0, to_page=0)

    # 构建输出文件的路径和名称，格式为"输出文件夹路径/page_{页码号}.pdf"

    # 保存新PDF文档到指定的输出文件路径
    output_pdf.save(output_folder)

    # 关闭新创建的PDF文档，释放资源
    output_pdf.close()
    # 关闭源PDF文件，释放资源
    source_pdf.close()

# 指定输入文件的路径为桌面上的一个PDF文件
input_file = "（老高考）24届高三押题卷·理综3.26.pdf"
# 指定输出文件夹的路径为桌面上的一个文件夹
output_folder = f"./（老高考）24届高三押题卷·理综3.26(1).pdf"
# 调用split_pdf函数，传入输入文件和输出文件夹的路径，开始执行PDF分割操作
split_pdf(input_file, output_folder)
